/*
题目: 可能的二分法
给定一组 n 人（编号为 1, 2, ..., n）， 我们想把每个人分进任意大小的两组。每个人都可能不喜欢其他人，那么他们不应该属于同一组。

给定整数 n 和数组 dislikes ，其中 dislikes[i] = [ai, bi] ，表示不允许将编号为 ai 和  bi的人归入同一组。当可以用这种方法将所有人分进两组时，返回 true；否则返回 false。

https://leetcode.cn/problems/possible-bipartition/
 */
public class PossibleBipartition {
    // 并查集
    int[] fa = null;
    int size ;

    public boolean possibleBipartition(int n, int[][] dislikes) {
        size = n;
        fa = new int[size + 1];
        Arrays.fill(fa, -1);
        List<Integer>[] group = new List[n + 1];
        for (int i = 0; i <= n; i ++) {
            group[i] = new ArrayList<> ();
        }

        for (int[] cur : dislikes) {
            group[cur[0]].add(cur[1]);
            group[cur[1]].add(cur[0]);
        }

        for (int i = 1; i <= n; i ++) {
            for (int j = 0; j < group[i].size(); j ++) {
                // 逐个合并
                unit(group[i].get(0), group[i].get(j));
                // 再进行逐个检查
                if (isConnected(i, group[i].get(j))) {
                    return false;
                }
            }
        }
        return true;
    }

    private int getRoot(int x) {
        return fa[x] >= 0 ? getRoot(fa[x]) : x;
    }

    private void unit(int x, int y) {
        if (x == y) return ; // 如果是同一个对象,就不合并了
        int idx1 = getRoot(x), idx2 = getRoot(y);
        if (idx1 == idx2)   return ;

        fa[idx1] += fa[idx2];
        fa[idx2] = idx1;
    }

    private boolean isConnected(int x, int y) {
        return getRoot(x) == getRoot(y);
    }
}
